from pydantic import BaseModel, Field


class UserLLMForm(BaseModel):
    id: str = Field(default=None, description="模型ID")
    chatModel: str = Field(default=None, description="聊天模型")
    embeddingModel: str = Field(default=None, description="embedding模型")
    img2txtModel: str = Field(default=None, description="img2txt模型")
    sequence2txtModel: str = Field(default=None, description="sequence2txt模型")
    rerankModel: str = Field(default=None, description="rerank模型")
    ttsModel: str = Field(default=None, description="tts模型")